package com.reactim.message.task;

import com.reactim.message.protection.ResourceProtectionService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

/**
 * 资源保护定时任务
 * 负责清理过期数据和监控资源使用情况
 */
@Component
@Slf4j
@RequiredArgsConstructor
public class ResourceProtectionTask {

    private final ResourceProtectionService resourceProtectionService;

    /**
     * 定期清理过期数据
     * 每5分钟执行一次
     */
    @Scheduled(fixedDelayString = "${resource.protection.cleanup-interval-ms:300000}")
    public void cleanupExpiredData() {
        try {
            log.debug("开始清理资源保护过期数据");
            resourceProtectionService.cleanupExpiredData();
            log.debug("资源保护过期数据清理完成");
            
        } catch (Exception e) {
            log.error("清理资源保护过期数据失败", e);
        }
    }

    /**
     * 定期监控资源使用情况
     * 每分钟执行一次
     */
    @Scheduled(fixedDelayString = "${resource.protection.monitor-interval-ms:60000}")
    public void monitorResourceUsage() {
        try {
            var stats = resourceProtectionService.getResourceUsageStats();
            
            // 记录关键指标
            log.info("资源使用监控 - 全局连接数: {}, 活跃用户数: {}, 活跃IP数: {}, 连接拒绝率: {:.2f}%", 
                stats.getGlobalConnectionCount(),
                stats.getActiveUserCount(),
                stats.getActiveIpCount(),
                stats.getConnectionRejectionRate() * 100);
            
            // 检查是否有异常情况
            if (stats.getConnectionRejectionRate() > 0.1) { // 拒绝率超过10%
                log.warn("连接拒绝率过高: {:.2f}%, 总创建: {}, 总拒绝: {}", 
                    stats.getConnectionRejectionRate() * 100,
                    stats.getTotalConnectionsCreated(),
                    stats.getTotalConnectionsRejected());
            }
            
            if (stats.getMaxUserConnections() > 8) { // 单用户连接数接近限制
                log.warn("检测到用户连接数过高: 最大用户连接数={}", stats.getMaxUserConnections());
            }
            
            if (stats.getMaxIpConnections() > 80) { // 单IP连接数接近限制
                log.warn("检测到IP连接数过高: 最大IP连接数={}", stats.getMaxIpConnections());
            }
            
        } catch (Exception e) {
            log.error("监控资源使用情况失败", e);
        }
    }
}